package diagonalPrime;

public class Solution {
    public int diagonalPrime(int[][] nums) {
        int max=0;
        for (int i = 0; i < nums.length; i++) {
            if(check(nums[i][i])){
                if(nums[i][i]>max){
                    max=nums[i][i];
                }
            }
        }
        if(max==0)
        for (int i = 0; i< nums.length  ; i++) {
            if(check(nums[i][nums.length-i-1])){
                if(nums[i][nums.length-1-i]>max){
                    max=nums[i][nums.length-1-i];
                }
            }
        }
        return max;
    }
    private boolean check(int n){
        if(n==1){
            return false;
        }
        int k = (int)Math.sqrt(n);
        for (int i = 2; i <= k; i++) {
            if(n%k==0){
                return false;
            }
        }
        return true;
    }
}
